from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import JsonResponse


class LoginRequiredJSONMixin(LoginRequiredMixin):
    """自定义LoginRequiredMixin 判断用户是否登录
    如果用户未登录,响应json,状态码400
    """

    def handle_no_permission(self):
        return JsonResponse({
            'code': 400,
            'errmsg': '用户未登录'
        })
